package com.leecode.hotCode;

import java.util.*;

/**
 * @Author orange7u7
 * @Date 2024/1/17 15:29
 * @PackageName:com.leecode.hotCode
 * @ClassName: GroupAnagrams
 * @Description: TODO
 * @Version 1.0
 */
public class GroupAnagrams {
    /**
     * 输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
     * 输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
     */
    /**
     * 由于互为字母异位词的两个字符串包含的字母相同，
     * 因此对两个字符串分别进行排序之后得到的字符串一定是相同的，
     * 故可以将排序之后的字符串作为哈希表的键。
     * @param args
     */
//    public List<List<String>> groupAnagrams(String[] strs) {
//        HashMap<String, List<String>> hashMap = new HashMap<String, List<String>>();
//        for (int i = 0; i < strs.length; i++) {
//            char[] charArray = strs[i].toCharArray();
//            Arrays.sort(charArray);
//            String sortedStr = new String(charArray);
//            List<String> stringList;
//            if (hashMap.containsKey(sortedStr)) {
//                stringList = hashMap.get(sortedStr);
//            } else {
//                stringList = new ArrayList<String>();
//            }
//            stringList.add(strs[i]);
//            hashMap.put(sortedStr, stringList);
//        }
//        ArrayList<List<String>> targetList = new ArrayList<List<String>>(hashMap.values());
//        return targetList;
//    }

    /**
     * 由于互为字母异位词的两个字符串包含的字母相同，因此两个字符串中的相同字母出现的次数一定是相同的，
     * 故可以将每个字母出现的次数使用字符串表示，作为哈希表的键。
     * 由于字符串只包含小写字母，因此对于每个字符串，可以使用长度为 26 的数组记录每个字母出现的次数。
     * @return
     */
    public List<List<String>> groupAnagrams(String[] strs) {
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        for (String str : strs) {
            int[] counts = new int[26];
            int length = str.length();
            for (int i = 0; i < length; i++) {
                counts[str.charAt(i) - 'a']++;
            }
            // 将每个出现次数大于 0 的字母和出现次数按顺序拼接成字符串，作为哈希表的键
            StringBuffer sb = new StringBuffer();
            for (int i = 0; i < 26; i++) {
                if (counts[i] != 0) {
                    sb.append((char) ('a' + i));
                    sb.append(counts[i]);
                }
            }
            String key = sb.toString();
            List<String> list = map.getOrDefault(key, new ArrayList<String>());
            list.add(str);
            map.put(key, list);
        }
        return new ArrayList<List<String>>(map.values());
    }

    public static void main(String[] args) {
        GroupAnagrams groupAnagrams = new GroupAnagrams();
        String[] strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
        groupAnagrams.groupAnagrams(strs);
    }
}
